from .mask_generators import NODE_CLASS_MAPPINGS as MASK_GENERATOR_NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as MASK_GENERATOR_DISPLAY_NAME_MAPPINGS
from .image_tools import NODE_CLASS_MAPPINGS as IMAGE_TOOL_NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as IMAGE_TOOL_DISPLAY_NAME_MAPPINGS
from .bbox_tools import NODE_CLASS_MAPPINGS as BBOX_TOOL_NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as BBOX_TOOL_DISPLAY_NAME_MAPPINGS
from .point_tools import NODE_CLASS_MAPPINGS as POINT_TOOL_NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as POINT_TOOL_DISPLAY_NAME_MAPPINGS
from .mask_tools import NODE_CLASS_MAPPINGS as MASK_TOOL_NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS as MASK_TOOL_DISPLAY_NAME_MAPPINGS


NODE_CLASS_MAPPINGS = {
    **MASK_GENERATOR_NODE_CLASS_MAPPINGS,
    **IMAGE_TOOL_NODE_CLASS_MAPPINGS,
    **BBOX_TOOL_NODE_CLASS_MAPPINGS,
    **POINT_TOOL_NODE_CLASS_MAPPINGS,
    **MASK_TOOL_NODE_CLASS_MAPPINGS,
}

NODE_DISPLAY_NAME_MAPPINGS = {
    **MASK_GENERATOR_DISPLAY_NAME_MAPPINGS,
    **IMAGE_TOOL_DISPLAY_NAME_MAPPINGS,
    **BBOX_TOOL_DISPLAY_NAME_MAPPINGS,
    **POINT_TOOL_DISPLAY_NAME_MAPPINGS,
    **MASK_TOOL_DISPLAY_NAME_MAPPINGS,
}


WEB_DIRECTORY = "./js"
CSS_DIRECTORY = "./css"

__all__ = [
    "NODE_CLASS_MAPPINGS",
    "NODE_DISPLAY_NAME_MAPPINGS",
    "WEB_DIRECTORY",
    "CSS_DIRECTORY"
]